Perfect Square in Python